import cv2
import numpy as np
from sightseeing_infer.cartoon.cartoon import Cartoon

class Ensemble:
    def __init__(self,cartoon_onnx_512):
        
        self.sightseeing_style = Cartoon(cartoon_onnx_512)
        
        
    def run(self,image_path):
        image = cv2.imread(image_path)
        image_bg = self.sightseeing_style.run(image,512)
        
        return image_bg
    
    
if __name__ == "__main__":
    pass